package com.coursehub.course.repository;

import com.coursehub.course.entity.ClassesStudent;
import com.coursehub.course.entity.ClassesStudentId;
import org.springframework.data.jpa.repository.JpaRepository;
import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;

import java.util.List;
import java.util.Optional;

/**
 * 班级学生关联数据访问层
 */
@Repository
public interface ClassesStudentRepository extends JpaRepository<ClassesStudent, ClassesStudentId> {
    
    /**
     * 根据班级ID查找所有班级学生关联
     */
    List<ClassesStudent> findByClasses_ClassesId(String classesId);
    
    /**
     * 根据用户ID查找所有班级学生关联
     */
    List<ClassesStudent> findById_UserId(String userId);
    
    /**
     * 根据班级ID和用户ID查找班级学生关联
     */
    Optional<ClassesStudent> findById_ClassesIdAndId_UserId(String classesId, String userId);
    
    /**
     * 检查学生是否在该班级中
     */
    boolean existsById_ClassesIdAndId_UserId(String classesId, String userId);
    
    /**
     * 检查班级学生关联是否存在（自定义查询）
     */
    @Query("SELECT EXISTS(SELECT cs FROM ClassesStudent cs WHERE cs.id.classesId = :classesId AND cs.id.userId = :userId)")
    boolean existsByClassesIdAndUserId(@Param("classesId") String classesId, @Param("userId") String userId);
    
    /**
     * 根据班级ID获取所有学生ID
     */
    @Query("SELECT cs.id.userId FROM ClassesStudent cs WHERE cs.id.classesId = :classesId")
    List<String> findStudentIdsByClassesId(@Param("classesId") String classesId);
    
    /**
     * 根据用户ID获取所有班级ID
     */
    @Query("SELECT cs.id.classesId FROM ClassesStudent cs WHERE cs.id.userId = :userId")
    List<String> findClassesIdsByUserId(@Param("userId") String userId);
}